import os
import json

from audio_processor import AudioProcessor


def main():
    """测试音频处理功能"""
    # 创建音频处理器实例
    processor = AudioProcessor(language="zh-CN")
    
    # 测试单个文件
    test_file = "static/media/audio/舌尖大叔-01.wav"  # 替换为你的测试音频文件
    
    if os.path.exists(test_file):
        print(f"\n处理单个音频文件: {test_file}")
        result = processor.extract_text(test_file)
        print(json.dumps(result, indent=2, ensure_ascii=False))
    else:
        print(f"测试文件不存在: {test_file}")
        print("请将测试文件路径替换为实际存在的音频文件")
    
    # 测试批量处理
    # test_dir = "static/media/audio"  # 替换为你的测试音频目录
    
    # if os.path.exists(test_dir) and os.path.isdir(test_dir):
    #     print(f"\n批量处理音频文件目录: {test_dir}")
    #     results = processor.batch_extract(test_dir)
        
    #     # 只打印摘要信息，避免输出过多
    #     summary = {
    #         "success": results["success"],
    #         "directory": results["directory"],
    #         "total_files": results["total_files"],
    #         "successful_extractions": results["successful_extractions"]
    #     }
    #     print(json.dumps(summary, indent=2, ensure_ascii=False))
        
    #     # 打印第一个结果作为示例
    #     if results["results"]:
    #         print("\n第一个处理结果示例:")
    #         print(json.dumps(results["results"][0], indent=2, ensure_ascii=False))
    # else:
    #     print(f"测试目录不存在: {test_dir}")
    #     print("请将测试目录路径替换为实际存在的目录")

if __name__ == "__main__":
    main()